package no.awi.wpd.model;

import no.awi.wpd.unit.model.Profile;
import no.awi.wpd.unit.model.Unit;

public class UnitMother {

    public static Unit skullcrushers(int number) {
        Unit unit = createSingleFileUnit(number);
        unit.setMount(ProfileMother.juggernaught());
        unit.setProfile(ProfileMother.skullcrusher());
        unit.setName("Skullcrushers");
        return unit;
    }

    public static Unit mournfang(int number) {
        Unit unit = createSingleFileUnit(number);
        unit.setMount(ProfileMother.mournfangRider());
        unit.setProfile(ProfileMother.mournfang());
        unit.setName("Mournfang");
        return unit;
    }

    public static Unit createSingleFileUnit(int number) {
        return createUnit(number, number);
    }

    public static Unit createUnit(int number, int filesize) {
        Unit unit = new Unit();
        unit.setFileSize(filesize);
        unit.setModelCount(number);
        return unit;
    }

}
